#!/usr/bin/env python
# -*- coding:utf-8 -*-
from langchain_community.embeddings import ZhipuAIEmbeddings


class ZhiPuEmbedding:
    """
    ZhiPu Embedding
    """

    def __init__(self, api_key: str, model='embedding-3'):
        """
            Initialize of the ZhiPu LLM, platform location: https://open.bigmodel.cn/dev
        :param api_key: access key
        """
        self.embedding = ZhipuAIEmbeddings(api_key=api_key, model=model)
